package com.ossbar.utils.tool;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

import javax.imageio.ImageIO;

public class ImageVerificationCode {

	private int weight = 200; // 验证码图片的长和宽
	private int height = 50;
	private String text; // 用来保存验证码的文本内容
	private Random r = new Random(); // 获取随机数对象
	// private String[] fontNames = {"宋体", "华文楷体", "黑体", "微软雅黑", "楷体_GB2312"};
	// //字体数组
	// 字体数组
	private String[] fontNames = { "Georgia" };
	// 验证码数组
	private String codes = "23456789abcdefghjkmnopqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ";

	private ImageVerificationCode() {}
	private static ImageVerificationCode ivc = new ImageVerificationCode();
	/**
	 * 获取随机的颜色
	 *
	 * @return
	 */
	private Color randomColor() {
		int r = this.r.nextInt(225); // 这里为什么是225，因为当r，g，b都为255时，即为白色，为了好辨认，需要颜色深一点。
		int g = this.r.nextInt(225);
		int b = this.r.nextInt(225);
		return new Color(r, g, b); // 返回一个随机颜色
	}

	/**
	 * 获取随机字体
	 *
	 * @return
	 */
	private Font randomFont() {
		int index = r.nextInt(fontNames.length); // 获取随机的字体
		String fontName = fontNames[index];
		int style = r.nextInt(5); // 随机获取字体的样式，0是无样式，1是加粗，2是斜体，3是加粗加斜体
		int size = r.nextInt(10) + 24; // 随机获取字体的大小
		return new Font(fontName, style, size); // 返回一个随机的字体
	}

	/**
	 * 获取随机字符
	 *
	 * @return
	 */
	private char randomChar() {
		int index = r.nextInt(codes.length());
		return codes.charAt(index);
	}

	/**
	 * 画干扰线，验证码干扰线用来防止计算机解析图片
	 *
	 * @param image
	 */
	private void drawLine(BufferedImage image) {
		Graphics2D g = (Graphics2D) image.getGraphics();
		for (int i = 0; i < 6; i++) {
			int x1 = r.nextInt(weight);
			int y1 = r.nextInt(height);
			int x2 = r.nextInt(weight);
			int y2 = r.nextInt(height);
			g.setColor(randomColor());
			g.drawLine(x1, y1, x2, y2);
		}
	}

	/**
	 * 创建图片的方法
	 *
	 * @return
	 */
	private BufferedImage createImage() {
		// 创建图片缓冲区
		BufferedImage image = new BufferedImage(weight, height, BufferedImage.TYPE_INT_RGB);
		// 获取画笔
		Graphics2D g = (Graphics2D) image.getGraphics();
		// 设置背景色随机
		g.setColor(new Color(255, 255, r.nextInt(245) + 10));
		g.fillRect(0, 0, weight, height);
		// 返回一个图片
		return image;
	}

	/**
	 * 获取验证码图片的方法
	 *
	 * @return
	 */
	public static BufferedImage getImage() {
		BufferedImage image = ivc.createImage();
		Graphics2D g = (Graphics2D) image.getGraphics(); // 获取画笔
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < 5; i++) // 画四个字符即可
		{
			String s = ivc.randomChar() + ""; // 随机生成字符，因为只有画字符串的方法，没有画字符的方法，所以需要将字符变成字符串再画
			sb.append(s); // 添加到StringBuilder里面
			float x = i * 1.0F * ivc.weight / 10 + 50; // 定义字符的x坐标
			g.setFont(ivc.randomFont()); // 设置字体，随机
			g.setColor(ivc.randomColor()); // 设置颜色，随机
			g.drawString(s, x, ivc.height / 3 * 2);
		}
		ivc.text = sb.toString();
		ivc.drawLine(image);
		return image;
	}

	/**
	 * 获取验证码文本的方法
	 *
	 * @return
	 */
	public static String getText() {
		return ivc.text;
	}

	public static void output(BufferedImage image, OutputStream out) throws IOException // 将验证码图片写出的方法
	{
		ImageIO.write(image, "png", out);
	}
}
